"""
    嵌入一幅二值图片, tiff格式。
    输入:
        隐藏载体路径
        需要隐藏的二值图片路径
    输出：
        隐藏图片以后的图片
"""
from cv2  import cv2
import sys
import hide

def printHelp():
    print("Help Usage：python {} [水印载体路径] [需要隐藏的黑白图片路径] [输出图片文件名.jpg]".format(sys.argv[0]))


if len(sys.argv)!=4 :
    printHelp()
    sys.exit()

if __name__=="__main__":
    printHelp()

PicCovered=sys.argv[1] # 载体
PicToHide=sys.argv[2] # 需要隐藏的数据
jpeg=hide.jio.read(PicCovered)
PicToHide=cv2.imread(PicToHide)


std=128 #  阈值。像素值 <128为黑色,>=128为白色

message=""
print("Debug:输入的图片规模{}".format(PicToHide.shape))
for i in range(0,PicToHide.shape[0]):
    for j in range(0,PicToHide.shape[1]):
        if PicToHide[i][j][0] < std:
            message +="0"
        else :
            message +="1"
print("Debug:Message Length:{}".format(len(message)))

key,jpeg=hide.imbed(jpeg,message)
print("Your Key:{}".format(key))
hide.jio.write(jpeg,sys.argv[3])
print("### {} Finish".format(PicCovered))
